import muggle_ocr

# ModelType.OCR 可识别光学印刷文本，但是只能单行文字，并且排版需要非常好（不能有很多空白）
sdk = muggle_ocr.SDK(model_type=muggle_ocr.ModelType.OCR)
with open(r"captcha1.png", "rb") as f:
    image_binary = f.read()
    text = sdk.predict(image_bytes=image_binary)
    print(text)


# ModelType.Captcha 可识别4-6位验证码，使用的训练模型，很精准，但是不能识别大段文字
sdk = muggle_ocr.SDK(model_type=muggle_ocr.ModelType.Captcha)
with open(r"captcha1.png", "rb") as f:
    image_binary = f.read()
    text = sdk.predict(image_bytes=image_binary)
    print(text)